#pragma once

#include <iostream>
#include <string>
#include <vector>
#include <atomic>
#include <fstream>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/time.h>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    class TimeUtil
    {
    public:
        // 获得当前时间戳（秒级别）
        static std::string GetTimeStamp()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec);
        }
        // 获取当前时间戳（毫秒级）
        static std::string GetTimeMs()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec * 1000 + _time.tv_usec / 1000);
        }
    };

    const std::string Path = "./temp/";
    class PathUtil
    {
    public:
        // 拼接字符串
        static std::string Addsuix(const std::string &file_name, const std::string &suffix)
        {
            std::string s = Path;
            s += file_name;
            s += suffix;
            return s;
        }
        // 生成 ./temp/xxxx.cc文件
        static std::string Src(const std::string &file_name)
        {
            return Addsuix(file_name, ".cc");
        }
        // 生成 ./temp/xxxx.exe可运行程序
        static std::string Exe(const std::string &file_name)
        {
            return Addsuix(file_name, ".exe");
        }

        // 生成 编译错误文件
        static std::string CompilerError(const std::string &file_name)
        {
            return Addsuix(file_name, ".compile_error");
        }

        // 运行时需要的临时文件
        static std::string Stdin(const std::string &file_name)
        {
            return Addsuix(file_name, ".stdin");
        }
        static std::string Stdout(const std::string &file_name)
        {
            return Addsuix(file_name, ".stdout");
        }
        // 构建该程序对应的标准错误完整的路径+后缀名
        static std::string Stderr(const std::string &file_name)
        {
            return Addsuix(file_name, ".stderr");
        }
    };

    class FileUtil
    {
    public:
        // 判断文件是否存在
        static bool IsFileExist(const std::string &file_name)
        {
            struct stat st;
            if (stat(file_name.c_str(), &st) == 0) // stat是一个取得文件属性的方法，返回值为0表示读取成功
            {
                return true;
            }
            return false;
        }
        // 使用毫秒级时间戳和原子性计数器来形成一个新的字符串
        static std::string UniqueFilename()
        {
            static std::atomic_uint id(0); // c++11特有的原子性计数器
            id++;
            std::string ms = TimeUtil::GetTimeMs();
            std::string uid = std::to_string(id);
            return ms + "_" + uid;
        }
        // 向指定文件写入字符串
        static bool WriteFile(const std::string &file_name, const std::string &content)
        {
            std::ofstream out(file_name);
            if (!out.is_open())
            {
                return false;
            }
            out.write(content.c_str(), content.size());
            out.close();
            return true;
        }
        // 将指定文件的内容读取到字符串
        static bool ReadFile(const std::string &file_name, std::string *content, bool keep = false)
        {
            (*content).clear();

            std::ifstream in(file_name);
            if (!in.is_open())
            {
                return false;
            }

            std::string line; // 表示每行读出来的字符串

            // 由于getline会跳过'\n'进行读取，用keep是否为true来标定是否需要读取'\n'
            while (std::getline(in, line))
            {
                (*content) += line;
                (*content) += (keep ? "\n" : ""); // 为true就每读出一行加上'\n'
            }
            in.close();
            return true;
        }
    };
    class StringUtil
    {
        // 用boost库进行字符串切割，以指定字符进行切割，结果存在vector中
    public:
        static void SplitString(const std::string &src, std::vector<std::string> *target,
                                const std::string &sep)
        {
            boost::split((*target), src, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };
}